from fastapi import APIRouter
from app.testapi.D1ChatTest.D1ChatTestController import router as d1chat_test_router
from app.testapi.D1ChatTest.D1LangChainCommunityController import router as d1langchain_community_router
from app.testapi.D1ChatTest.D1OpenAIController import router as d1openai_router
from app.testapi.D2Prompt.PromptOllamaController import router as d2prompt_ollama_router
from app.testapi.D2Prompt.systemController import router as system_router
from app.testapi.D2Prompt.UserRoleController import router as user_role_router
from app.testapi.D2Prompt.assistantRoleController import router as assistant_role_router
from app.testapi.D2Prompt.CombinePipeLineController import router as combine_pipe_line_router
from app.testapi.D3PromptTemplate.PromptTemplateD1 import router as prompt_template_d1_router
from app.testapi.D4RAG.RAGControllerD1 import router as rag_controller_d1_router
from app.testapi.D4RAG.RAGControllerD2 import router as rag_controller_d2_router
from app.testapi.D5LLMs.OllamaLiuShiController import router as ollama_liushi_router
from app.testapi.D5LLMs.ConversationBufferMemory import router as conversation_buffer_memory_router
from app.testapi.DTest.DTest import router as dtest_router

testapi_router = APIRouter(prefix="/testapi")

testapi_router.include_router(d1chat_test_router,prefix="/d1chat",tags=["D1ChatTest"])
testapi_router.include_router(d1langchain_community_router,prefix="/d1langchain",tags=["D1LangChainCommunity"])
testapi_router.include_router(d1openai_router,prefix="/d1openai",tags=["D1OpenAI"])
testapi_router.include_router(d2prompt_ollama_router,prefix="/d2prompt",tags=["D2Prompt"])
testapi_router.include_router(system_router,prefix="/system",tags=["system"])
testapi_router.include_router(user_role_router,prefix="/user_role",tags=["user_role"])
testapi_router.include_router(assistant_role_router,prefix="/assistant_role",tags=["assistant_role"])
testapi_router.include_router(combine_pipe_line_router,prefix="/combine_pipe_line",tags=["combine_pipe_line"])
testapi_router.include_router(prompt_template_d1_router,prefix="/prompt_template_d1",tags=["prompt_template_d1"])
testapi_router.include_router(rag_controller_d1_router,prefix="/rag_controller_d1",tags=["rag_controller_d1"])
testapi_router.include_router(rag_controller_d2_router,prefix="/rag_controller_d2",tags=["rag_controller_d2"])
testapi_router.include_router(ollama_liushi_router,prefix="/ollama_liushi",tags=["ollama_liushi"])
testapi_router.include_router(conversation_buffer_memory_router,prefix="/conversation_buffer_memory",tags=["conversation_buffer_memory"])
testapi_router.include_router(dtest_router,prefix="/dtest",tags=["dtest"])
